#include <iostream>
#include <vector>
#include <string>
#include <sw/redis++/redis++.h>

#include "util.hpp"

void test1(sw::redis::Redis& redis)
{
    std::cout << "hset 和 hget" << std::endl;
    redis.flushall();

    // 第一种插入方法, 直接插入
    redis.hset("key", "f1", "111");

    // 第二种插入方法, pair
    redis.hset("key", std::make_pair("f2", "222"));

    // 第三种插入方法, 通过容器迭代器插入
    std::vector<std::pair<std::string, std::string>> fields = {
        std::make_pair("f3", "333"),
        std::make_pair("f4", "444")
    };
    redis.hset("key", fields.begin(), fields.end());

    // 第四种插入方法, 通过初始化列表方式插入
    redis.hset("key", {
        std::make_pair("f5", "555"),
        std::make_pair("f6", "666")
    });

    sw::redis::OptionalString result = redis.hget("key", "f100");
    if (result)
    {
        std::cout << "result: " << result.value() << std::endl;
    }
    else
    {
        std::cout << "无效值" << std::endl;
    }
}

void test2(sw::redis::Redis& redis)
{
    std::cout << "hexists" << std::endl;
    redis.flushall();

    redis.hset("key", "f1", "111");
    redis.hset("key", "f2", "222");
    redis.hset("key", "f3", "333");

    bool result = redis.hexists("key", "f4");
    std::cout << "result: " << result << std::endl;
}

void test3(sw::redis::Redis& redis)
{
    std::cout << "hdel" << std::endl;
    redis.flushall();

    redis.hset("key", {
        std::make_pair("f1", "111"),
        std::make_pair("f2", "222"),
        std::make_pair("f3", "333"),
        std::make_pair("f4", "444"),
        std::make_pair("f5", "555"),
        std::make_pair("f6", "666"),
        std::make_pair("f7", "777")
    });

    long long result = redis.hdel("key", "f1");
    std::cout << "result: " << result << std::endl;

    result = redis.hdel("key", {"f2", "f3"});
    std::cout << "result: " << result << std::endl;

    std::vector<std::string> fields = {
        "f4", "f5", "f6"
    };
    result = redis.hdel("key", fields.begin(), fields.end());
    std::cout << "result: " << result << std::endl;

    long long len = redis.hlen("key");
    std::cout << "len: " << len << std::endl;
}

void test4(sw::redis::Redis& redis)
{
    std::cout << "hkeys 和 hvals" << std::endl;
    redis.flushall();

    redis.hset("key", "f1", "111");
    redis.hset("key", "f2", "222");
    redis.hset("key", "f3", "333");

    std::vector<std::string> fields;
    auto fieldsIt = std::back_inserter(fields);
    redis.hkeys("key", fieldsIt);
    printContainer(fields);

    std::vector<std::string> values;
    auto valuesIt = std::back_inserter(values);
    redis.hvals("key", valuesIt);
    printContainer(values);
}

void test5(sw::redis::Redis& redis)
{
    std::cout << "hmget 和 hmset" << std::endl;
    redis.flushall();

    redis.hmset("key", {
        std::make_pair("f1", "111"),
        std::make_pair("f2", "222")
    });

    std::vector<std::pair<std::string, std::string>> pairs = {
        {"f3", "333"},
        {"f4", "444"}
    };
    redis.hmset("key", pairs.begin(), pairs.end());

    std::vector<std::string> results;
    auto it = std::back_inserter(results);
    redis.hmget("key", {"f1", "f2", "f3", "f4"}, it);

    printContainer(results);
}

int main()
{
    sw::redis::Redis redis("tcp://127.0.0.1:6379");

    // test1(redis);
    // test2(redis);
    // test3(redis);
    // test4(redis);
    test5(redis);

    return 0;
}